# Part of Odoo. See LICENSE file for full copyright and licensing details.

from odoo import api, fields, models, _


class RepairOrder(models.Model):
    _inherit = 'repair.order'

    production_count = fields.Integer(
        'Count of MOs generated',
        compute='_compute_production_count',
        groups='mrp.group_mrp_user',
    )

    @api.depends('reference_ids.production_ids')
    def _compute_production_count(self):
        for repair in self:
            repair.production_count = len(repair.reference_ids.production_ids)

    @api.model_create_multi
    def create(self, vals_list):
        orders = super().create(vals_list)
        orders.action_explode()
        return orders

    def write(self, vals):
        res = super().write(vals)
        self.action_explode()
        return res

    def action_explode(self):
        lines_to_unlink_ids = set()
        line_vals_list = []
        for op in self.move_ids:
            bom = self.env['mrp.bom'].sudo()._bom_find(op.product_id, company_id=op.company_id.id, bom_type='phantom')[op.product_id]
            if not bom:
                continue
            factor = op.product_uom._compute_quantity(op.product_uom_qty, bom.product_uom_id) / bom.product_qty
            _boms, lines = bom.sudo().explode(op.product_id, factor, picking_type=bom.picking_type_id)
            for bom_line, line_data in lines:
                if bom_line.product_id.type != 'service':
                    line_vals_list.append(op._prepare_phantom_line_vals(bom_line, line_data['qty']))
            lines_to_unlink_ids.add(op.id)

        self.env['stock.move'].browse(lines_to_unlink_ids).sudo().unlink()
        if line_vals_list:
            self.env['stock.move'].create(line_vals_list)

    def action_view_mrp_productions(self):
        self.ensure_one()
        production_order_ids = self.reference_ids.production_ids
        action = {
            'type': 'ir.actions.act_window',
            'res_model': 'mrp.production',
            'views': [[False, 'form']],
        }

        if self.production_count == 1:
            action['res_id'] = production_order_ids.id
        elif self.production_count > 1:
            action['name'] = _("Manufacturing Orders generated by %s", self.name)
            action['views'] = [[False, 'list']]
            action['domain'] = [('id', 'in', production_order_ids.ids)]

        return action

    def _get_action_add_from_catalog_extra_context(self):
        bom = self.env['mrp.bom']._bom_find(self.product_id, company_id=self.company_id.id)[self.product_id]
        product_ids = [line.product_id.id for line in bom.bom_line_ids] if bom else []
        return {
            **super()._get_action_add_from_catalog_extra_context(),
            'catalog_bom_product_ids': product_ids,
            'search_default_bom_parts': bool(product_ids)
        }


class StockMove(models.Model):
    _inherit = 'stock.move'

    def _prepare_phantom_line_vals(self, bom_line, qty):
        self.ensure_one()
        product = bom_line.product_id
        return {
            'repair_id': self.repair_id.id,
            'repair_line_type': self.repair_line_type,
            'product_id': product.id,
            'price_unit': self.price_unit,
            'product_uom_qty': qty,
            'location_id': self.location_id.id,
            'location_dest_id': self.location_dest_id.id,
            'state': 'draft',
        }
